from deepface import DeepFace
import time

# 寻找照片对应的人

# 输入待识别的人脸图像
img_path = "person1_1.jpg"

# 记录开始时间
start_time = time.time()
# 从数据库中匹配
result = DeepFace.find(
    img_path=img_path,
    db_path="database/",  # 数据库路径
    model_name="OpenFace",  # 推荐用 Facenet 提升精度
    detector_backend="mtcnn"
)

# 记录结束时间并计算耗时
end_time = time.time()
search_time = end_time - start_time

# 输出匹配结果（按相似度排序，最相似的在前）
print("最相似的匹配：")
print(result[0][["identity", "distance"]])  # identity 是匹配到的图像路径，distance 是距离

print(f"搜索耗时: {search_time:.2f} 秒")